# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
try:
    from .backbone import (
        build_vgg_backbone,
        build_ws_resnet_backbone,
    )
except:
    from .backbone import (
        build_vgg_backbone,
    )

from .postprocessing import detector_postprocess

try:
    from .roi_heads import WSDDNROIHeads, CSCROIHeads, OICRROIHeads, PCLROIHeads
except:
    from .roi_heads import WSDDNROIHeads, OICRROIHeads
    
try:
    from .roi_heads import UWSODROIHeads
except:
    pass

try:
    from .seg_heads import WSJDSROIHeads
except:
    pass

from .test_time_augmentation_avg import DatasetMapperTTAAVG, GeneralizedRCNNWithTTAAVG
from .test_time_augmentation_union import DatasetMapperTTAUNION, GeneralizedRCNNWithTTAUNION

_EXCLUDE = {"torch", "ShapeSpec"}
__all__ = [k for k in globals().keys() if k not in _EXCLUDE and not k.startswith("_")]
